package com.dxy.netty.websocket;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;

import java.time.LocalDateTime;

/**
 * @Description 基于netty的WebSocket服务器处理器
 * @Author dangxianyue
 * @Date 2023/1/6 16:07
 */
public class NettyWebSocketServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {

    // 读取客户端发送的请求报文
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
        System.out.println("服务器端收到消息 = " + msg.text());

        //发给所有人
        //注意此处一定得是new TextWebSocketFrame()
        //参考：https://blog.csdn.net/zhengjcken/article/details/103313936
        ChannelHolder.send2All(new TextWebSocketFrame(msg.text()));

        // 给自己回复消息
        //ctx.channel().writeAndFlush(new TextWebSocketFrame(LocalDateTime.now() + "服务器回复:" + msg.text()));
    }

    // 当web客户端连接后，触发该方法
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        // ctx.channel().id() 表示唯一的值
        System.out.println("handlerAdded 被调用， channel.id.longText = " + ctx.channel().id().asLongText());
        System.out.println("handlerAdded 被调用， channel.id.shortText = " + ctx.channel().id().asShortText());
        ChannelHolder.addChannel(ctx.channel());
    }

    // 客户端离线
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        // ctx.channel().id() 表示唯一的值
        System.out.println("handlerRemoved 被调用， channel.id.longText = " + ctx.channel().id().asLongText());
        ChannelHolder.removeChannel(ctx.channel());
    }

    // 处理异常
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        System.out.println("异常发生，异常消息 = " + cause.getMessage());
        cause.printStackTrace();
        // 关闭连接
//        ctx.close();
        ctx.channel().close();
    }

}
